package utils

import (
	"net/url"
	"strings"
)

// HasHTTP 获取协议头
func HasHTTP(url string) bool {
	return strings.HasPrefix(url, "http://") || strings.HasPrefix(url, "https://")
}

// Wechat 判断是否是微信发起的请求
func Wechat(userAgent string) bool {
	return strings.Contains(strings.ToLower(userAgent), "micromessenger")
}

// GetQueryWithQuestionMark 获取参数参数
func GetQueryWithQuestionMark(rawURL string) string {
	parsedURL, err := url.Parse(rawURL)
	if err != nil {
		return "" // 或者根据需求处理错误
	}
	if parsedURL.RawQuery != "" {
		return "?" + parsedURL.RawQuery
	}
	return ""
}

// GetDomain 获取子域名名称
func GetDomain(domain string) string {
	return strings.Split(domain, ".")[0]
}
